package com.yanxiuhair.common.utils;

import java.util.Iterator;
import java.util.Set;
import org.apache.shiro.cache.Cache;
import org.apache.shiro.cache.CacheManager;
import org.apache.shiro.cache.ehcache.EhCacheManager;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import com.yanxiuhair.common.utils.spring.SpringUtils;

/**
 * @ClassName:  CacheUtils   
 * @Description: Cache工具类  
 * @author: gaoxiaochuang   
 * @date:   2021年7月1日 下午3:22:21   
 *     
 * @Copyright: 2021 http://www.yanxiuhair.com/ Inc. All rights reserved. 
 * 注意：本内容仅限于许昌妍秀发制品有限公司内部传阅，禁止外泄以及用于其他的商业目
 */
public class CacheUtils {
	private static Logger logger = LoggerFactory.getLogger(CacheUtils.class);

	private static CacheManager cacheManager = SpringUtils.getBean(CacheManager.class);

	private static final String SYS_CACHE = "sys-cache";

	/**
	 * 获取SYS_CACHE缓存
	 * 
	 * @param key
	 * @return
	 */
	public static Object get(String key) {
		return get(SYS_CACHE, key);
	}

	/**
	 * 获取SYS_CACHE缓存
	 * 
	 * @param key
	 * @param defaultValue
	 * @return
	 */
	public static Object get(String key, Object defaultValue) {
		Object value = get(key);
		return value != null ? value : defaultValue;
	}

	/**
	 * 写入SYS_CACHE缓存
	 * 
	 * @param key
	 * @return
	 */
	public static void put(String key, Object value) {
		put(SYS_CACHE, key, value);
	}

	/**
	 * 从SYS_CACHE缓存中移除
	 * 
	 * @param key
	 * @return
	 */
	public static void remove(String key) {
		remove(SYS_CACHE, key);
	}

	/**
	 * 获取缓存
	 * 
	 * @param cacheName
	 * @param key
	 * @return
	 */
	public static Object get(String cacheName, String key) {
		return getCache(cacheName).get(getKey(key));
	}

	/**
	 * 获取缓存
	 * 
	 * @param cacheName
	 * @param key
	 * @param defaultValue
	 * @return
	 */
	public static Object get(String cacheName, String key, Object defaultValue) {
		Object value = get(cacheName, getKey(key));
		return value != null ? value : defaultValue;
	}

	/**
	 * 写入缓存
	 * 
	 * @param cacheName
	 * @param key
	 * @param value
	 */
	public static void put(String cacheName, String key, Object value) {
		getCache(cacheName).put(getKey(key), value);
	}

	/**
	 * 从缓存中移除
	 * 
	 * @param cacheName
	 * @param key
	 */
	public static void remove(String cacheName, String key) {
		getCache(cacheName).remove(getKey(key));
	}

	/**
	 * 从缓存中移除所有
	 * 
	 * @param cacheName
	 */
	public static void removeAll(String cacheName) {
		Cache<String, Object> cache = getCache(cacheName);
		Set<String> keys = cache.keys();
		for (Iterator<String> it = keys.iterator(); it.hasNext();) {
			cache.remove(it.next());
		}
		logger.info("清理缓存： {} => {}", cacheName, keys);
	}

	/**
	 * 从缓存中移除指定key
	 * 
	 * @param keys
	 */
	public static void removeByKeys(Set<String> keys) {
		removeByKeys(SYS_CACHE, keys);
	}

	/**
	 * 从缓存中移除指定key
	 * 
	 * @param cacheName
	 * @param keys
	 */
	public static void removeByKeys(String cacheName, Set<String> keys) {
		for (Iterator<String> it = keys.iterator(); it.hasNext();) {
			remove(it.next());
		}
		logger.info("清理缓存： {} => {}", cacheName, keys);
	}

	/**
	 * 获取缓存键名
	 * 
	 * @param key
	 * @return
	 */
	private static String getKey(String key) {
		return key;
	}

	/**
	 * 获得一个Cache，没有则显示日志。
	 * 
	 * @param cacheName
	 * @return
	 */
	public static Cache<String, Object> getCache(String cacheName) {
		Cache<String, Object> cache = cacheManager.getCache(cacheName);
		if (cache == null) {
			throw new RuntimeException("当前系统中没有定义“" + cacheName + "”这个缓存。");
		}
		return cache;
	}

	/**
	 * 获取所有缓存
	 * 
	 * @return 缓存组
	 */
	public static String[] getCacheNames() {
		return ((EhCacheManager) cacheManager).getCacheManager().getCacheNames();
	}
}
